/*
 * Licensed to the Tai Ping Jin Ke
 *
 * Copyright (c) 2022 .
 * All rights reserved.
 * 项目名称：呼叫中心-网关层
 * 版权说明：本软件属太平金融科技服务(上海)有限公司所有，在未获得太平金融科技服务(上海)有限公司正式授权情况下，任何企业和个人，不能获取、阅读、安装、传播本软件涉及的任何受知识产权保护的内容。
 */
package com.taiping.cc.gateway.security.handler;

import com.taipingframework.boot.web.response.result.ApiResult;
import com.taipingframework.boot.web.response.status.ApiStatusEnum;
import com.taipingframework.json.JacksonHelper;
import lombok.extern.slf4j.Slf4j;
import org.springframework.core.io.buffer.DataBuffer;
import org.springframework.http.HttpStatus;
import org.springframework.http.MediaType;
import org.springframework.http.server.reactive.ServerHttpResponse;
import org.springframework.security.core.AuthenticationException;
import org.springframework.security.web.server.ServerAuthenticationEntryPoint;
import org.springframework.stereotype.Component;
import org.springframework.web.server.ServerWebExchange;
import reactor.core.publisher.Mono;

/**
 * 未认证处理，认证失败或抛异常时的处理程序。
 * 常见的认证方式包括但不限于 用户名/密码、手机号/验证码 等方式。
 */
@Slf4j
@Component
public class DefaultAuthenticationEntryPoint implements ServerAuthenticationEntryPoint {

    @Override
    public Mono<Void> commence(ServerWebExchange exchange, AuthenticationException exception) {
        return Mono.defer(() -> ApiResult.create()
                .doFirst(() -> log.error("发生{}异常！原因是：{}", exception.getClass(), exception.getMessage(), exception))
                .map(result -> result.setStatus(ApiStatusEnum.ERR_A0230))
                .map(JacksonHelper::to)
                .flatMap(message -> {
                    ServerHttpResponse response = exchange.getResponse();
                    response.setStatusCode(HttpStatus.OK);
                    // 此处需要注意的一点是，外部在后端调用本系统的接口时，
                    // 因为在此处修改了HTTP响应状态码，所以客户端无法收到响应报文，只会得到一个被抛出的异常。
                    // 基于这个考虑，此处依然需要返回200的HTTP状态码
                    //response.setStatusCode(HttpStatus.NETWORK_AUTHENTICATION_REQUIRED);
                    response.getHeaders().setContentType(MediaType.APPLICATION_JSON);
                    DataBuffer dataBuffer = response.bufferFactory().wrap(message.getBytes());
                    return response.writeWith(Mono.just(dataBuffer));
                }));
    }

}
